from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response


def jwt_response_payload_handler(token, user=None, request=None):
    """

    :param token:       认证成功之后，系统为我们生成的token
    :param user:        认证成功之后的user对象
    :param request:     登陆认证的请求
    :return:
    """
    return {
        'token': token,
        'username': user.username,
        'id': user.id
    }


class PageNum(PageNumberPagination):
    page_size = 6  # 每页个数
    # 想要使用 page size 就要设置 page_size_query_param
    page_size_query_param = 'pagesize'
    # 每页最大访问条数
    max_page_size = 20

    def get_paginated_response(self, data):
        # 返回的数据
        return Response({
            'count': self.page.paginator.count,  # 总数量
            'lists': data,  # 用户数据
            'page': self.page.number,  # 当前页数
            'pages': self.page.paginator.num_pages,  # 总页数
            'pagesize': self.page_size  # 后端指定的页容量

        })